pytorch-lightning>=1.3
torchvision
wandb
torchmetrics==0.4.1
torch>=1.6, <1.9
lightning-bolts
